"""
美国-日本-印度：GDP折线图
"""

"""
演示可视化需求1：折线图开发
"""

import json
from pyecharts.charts import Line
from pyecharts.options import (
    TitleOpts,
    LabelOpts,
    LegendOpts,
    ToolboxOpts,
    AxisPointerOpts,
)

# 处理数据
f_us = open("美国.txt", "r", encoding="UTF-8")
us_data = f_us.read()  # 美国的全部内容

f_jp = open("日本.txt", "r", encoding="UTF-8")
jp_data = f_jp.read()  # 日本的全部内容

f_in = open("印度.txt", "r", encoding="UTF-8")
in_data = f_in.read()  # 印度的全部内容

# 去掉不合JS0N规范的开头
us_data = us_data.replace("jsonp_1629344292311_69436(", "")
jp_data = jp_data.replace("jsonp_1629350871167_29498(", "")
in_data = in_data.replace("jsonp_1629350745930_63180(", "")

# 去掉不合JS0N规范的结尾
us_data = us_data[:-2]
jp_data = jp_data[:-2]
in_data = in_data[:-2]

# JS0N转Python字典
us_dict = json.loads(us_data)
jp_dict = json.loads(jp_data)
in_dict = json.loads(in_data)
# print(type(us_dict))
# print(us_dict)

# 获取trend key
us_trend_data = us_dict["data"][0]["trend"]
jp_trend_data = jp_dict["data"][0]["trend"]
in_trend_data = in_dict["data"][0]["trend"]
# print(type(trend_data))
# print(trend_data)

# 获取日期数据，用于x轴，取2020年(根据数据信息需要取到314)
# - 解释： trend_data['updateDate']表示将字典中关键字为'updateDate'的数据取出来   切片[:314]是将前面313个数据取出来
us_x_data = us_trend_data["updateDate"][:314]  # 查看文档后取314
jp_x_data = jp_trend_data["updateDate"][:315]  # 查看文档后取315
in_x_data = in_trend_data["updateDate"][:269]  # 查看文档后取269
# print(x_data)

# 获取确认数据，用于y轴，取2020年(到314下标结束)
us_y_data = us_trend_data["list"][0]["data"][:314]
jp_y_data = jp_trend_data["list"][0]["data"][:315]
in_y_data = in_trend_data["list"][0]["data"][:269]
# print(y_data)

# 生成图表
line = Line()  # 构建折线图对象
# 添X抽数据，数据都一样
line.add_xaxis(us_x_data)
# 添心抽数据
# label_opts=LabelOpts(is_show=False) 是指不显示标签上的数字，看起来就不会那么杂
line.add_yaxis(
    "美国确诊人数", us_y_data, label_opts=LabelOpts(is_show=False)
)  # 添加美国的y轴数摇    label_opts=LabelOpts(is_show=False)：标签不显示数字
line.add_yaxis(
    "日本确诊人数", jp_y_data, label_opts=LabelOpts(is_show=False)
)  # 添伽日本的y轴数据
line.add_yaxis(
    "印度确诊人数", in_y_data, label_opts=LabelOpts(is_show=False)
)  # 添加印度的y轴数据

# 设置全局选项
line.set_global_opts(
    # 标题设置
    title_opts=TitleOpts(
        title="2020年美日印三国确诊人数对比折线图", pos_left="center", pos_bottom="1%"
    ),
    legend_opts=LegendOpts(is_show=True),
    toolbox_opts=ToolboxOpts(is_show=True),
    axispointer_opts=AxisPointerOpts(),
)

# 调用render方法，生成图表
line.render()

# 关闭文件对象
f_us.close()
f_jp.close()
f_in.close()
